torch
torchvision
tensorly-torch
albumentations==0.4.6
datasets